# import argparse
#
# parser = argparse.ArgumentParser()
#
# # datasets options
# parser.add_argument('--data_path', type=str, default='/home/dataset/')
#
# # training options
# parser.add_argument('--ngpus', type=int, default=1)
#
# # loss options
# parser.add_argument('--hole', type=float, default=6)
# parser.add_argument('--valid', type=float, default=1)
# parser.add_argument('--style', type=float, default=120)
# parser.add_argument('--perceptual', type=float, default=0.05)
# parser.add_argument('--total_variation', type=float, default=0.1)
#
# opt = parser.parse_args()

import os

class Option(object):
    def __init__(self):

        # path
        self.image_path = ""
        self.train_image_path = ""
        self.test_image_path = ""
        self.mask_path = ""
        self.train_mask_path = ""
        self.test_mask_path = ""

        # batch_size
        self.train_batch_size = 6
        self.test_batch_size = 6
        self.n_threads = 4

        # check_point
        self.save_path = ""
        self.pretrain_model = ""
        self.check_point = ""

        # model
        self.n_gpus = 1
        self.gpu0 = 0

        # trainer
        self.adam_lr = 1
        self.decay_rate = 0.5
        self.ratio = [8, 15, 20, 25]
        self.momentum = 0.9
        self.weight_decay = 5e-4

        # training
        self.nepoch = 200
        self.threshold_num = 10000

        self.log_file = ""

        # Loss
        self.hole = 6
        self.valid = 1
        self.style = 120
        self.perceptual = 0.05
        self.total_variation = 0.1